#include <iostream>
#include <memory>
#include "Route.hpp"     // 路由的功能
#include "UdpServer.hpp" // 网络通信的功能
#include "ThreadPool.hpp"
#include "InetAddr.hpp"

using namespace ThreadPoolModule;
using task_t = std::function<void()>;

// ./udpserver port
int main(int argc, char *argv[])
{
    if (argc != 2)
    {
        std::cerr << "Usage: " << argv[0] << " port" << std::endl;
        return 1;
    }
    uint16_t port = std::stoi(argv[1]);

    Enable_Console_Log_Strategy();

    // 1、路由服务
    Route r;

    // 2、线程池
    auto tp = ThreadPool<task_t>::GetInstance();

    // 3、网络服务器对象，提供通信功能
    std::unique_ptr<UdpServer> usvr = std::make_unique<UdpServer>(port, [&r, &tp](int sockfd, const std::string &message, InetAddr &peer)  
    {
        // UdpServer的回调函数需要做的两件事
        // 构建任务。                                                        
        auto t = std::bind(&Route::MessageRoute, &r, sockfd, message, peer); // 这个函数参数为void，因为已经bind了固定参数
        // 任务入队列。之后就由线程池内的一个指定的线程执行路由功能。
        tp->Enqueue(t); 
    });
    

    // 单进程版
    // std::unique_ptr<UdpServer> usvr = std::make_unique<UdpServer>(port,[&r](int sockfd,const std::string &message,InetAddr &peer){
    //     r.MessageRoute(sockfd,message,peer);
    // } );

    usvr->Init();
    usvr->Start();
    return 0;
}
